from django.contrib import admin
from .models import *
# Register your models here.

class UserAdmin(admin.ModelAdmin):
    list_display = ('username', 'email', 'is_student', 'is_teacher')
    list_filter = ('is_student', 'is_teacher')
    search_fields = ('username', 'email')
admin.site.register(CustomUser, UserAdmin)



class InviteCodeAdmin(admin.ModelAdmin):
    list_display = ('code', 'created_time', 'used')
    search_fields = ('code',)
    list_filter = ('used',)
admin.site.register(InviteCodeModel, InviteCodeAdmin)